#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import argparse

from utils.dataset_utils import update_dataset


def parse_opt():
    parser = argparse.ArgumentParser()
    
    # ------------------------------------------------------------- #
    # PTL dataset options
    # ------------------------------------------------------------- #
    parser.add_argument('--exp_id', type=str, default='exp1')
    parser.add_argument('--cur_ptl_iter', type=str, default='iter_0')
    parser.add_argument('--min_bbox_hw_gan', type=int, default=10)
    parser.add_argument('--target_hw_gan', type=int, default=256)
    
    return parser.parse_args()   


def main(opt):
    # Dataset initialization
    update_dataset(opt)


if __name__ == "__main__":
    opt = parse_opt()
    main(opt)